import cv2
import numpy as np
from matplotlib import pyplot as plt
import random,time
from PIL import ImageGrab
import pyautogui
import cv2
import numpy as np
import time
from 子函数 import yinru
from 子函数.yinru import cs,zd ,yqfy,jr ,zb ,js ,sb ,yh ,ts ,bqds ,zd1 ,cjdw ,cj ,kszd ,qd ,jjtp ,jg ,yq ,di2zhang,di3zhang ,di4zhang ,di5zhang
from 子函数.yinru import di20zhang ,di21zhang ,di22zhang ,di23zhang ,ts1 ,daotubiao ,daotubiao1 ,zhanliping ,huodejiangli ,zaizhanchang ,haifangzhu ,sx ,tz ,xiaomao
from 子函数.yinru import jiejietupo ,liuxunzhang,jieshu,danjujieshu,qr1,时间下限,时间上限,feiliuxing,jieshou,time3,jingyanman,qx,douji,baoxiang,tanshouweizhi

def 当前阴阳师在什么位置():
    zhuangtai = 0
    img_rgb = np.array(ImageGrab.grab().convert('RGB'))#屏幕截图
    img_gray = cv2.cvtColor(img_rgb, cv2.COLOR_BGR2GRAY)

    # 结界突破 位置
    res = cv2.matchTemplate(img_gray, jiejietupo, cv2.TM_CCOEFF_NORMED)  # 大图 ，目标 ，模式
    threshold = 0.8  # 阈值 图片相识度
    loc = np.where(res >= threshold)
    # pyautogui.moveTo(random.randint(0, 1430), random.randint(0, 960))  # 鼠标移动到坐标位置
    for pt in zip(*loc[::-1]):

        print("结界突破 位置")
        zhuangtai = 1
        return zhuangtai


    # 战场位置 位置
    res = cv2.matchTemplate(img_gray, zaizhanchang, cv2.TM_CCOEFF_NORMED)  # 大图 ，目标 ，模式
    threshold = 0.8  # 阈值 图片相识度
    loc = np.where(res >= threshold)
    # pyautogui.moveTo(random.randint(0, 1430), random.randint(0, 960))  # 鼠标移动到坐标位置
    for pt in zip(*loc[::-1]):
        print("战场位置 位置")
        zhuangtai = 2
        return zhuangtai

    # 探索进入 位置
    res = cv2.matchTemplate(img_gray, tanshouweizhi, cv2.TM_CCOEFF_NORMED)  # 大图 ，目标 ，模式
    threshold = 0.8  # 阈值 图片相识度
    loc = np.where(res >= threshold)
    # pyautogui.moveTo(random.randint(0, 1430), random.randint(0, 960))  # 鼠标移动到坐标位置
    for pt in zip(*loc[::-1]):
        print("探索进入 位置")
        zhuangtai = 3
        return zhuangtai
    # 结束点罐子 位置
    res = cv2.matchTemplate(img_gray, js, cv2.TM_CCOEFF_NORMED)  # 大图 ，目标 ，模式
    threshold = 0.8  # 阈值 图片相识度
    loc = np.where(res >= threshold)
    # pyautogui.moveTo(random.randint(0, 1430), random.randint(0, 960))  # 鼠标移动到坐标位置
    for pt in zip(*loc[::-1]):
        pyautogui.moveTo(int(pt[0])+10, int(pt[1])+10)  # 鼠标移动到坐标位置
        pyautogui.click()  # 鼠标点击
        print("结束点罐子")
        zhuangtai = 4
        return zhuangtai

    # 失败 位置
    res = cv2.matchTemplate(img_gray, sb, cv2.TM_CCOEFF_NORMED)  # 大图 ，目标 ，模式
    threshold = 0.8  # 阈值 图片相识度
    loc = np.where(res >= threshold)
    # pyautogui.moveTo(random.randint(0, 1430), random.randint(0, 960))  # 鼠标移动到坐标位置
    for pt in zip(*loc[::-1]):
        pyautogui.moveTo(int(pt[0])+10, int(pt[1])+10)  # 鼠标移动到坐标位置
        pyautogui.click()  # 鼠标点击
        print("结束点罐子")
        zhuangtai = 5
        return zhuangtai

